package com.tencent.bk.utils.blueking.tools.excel;


import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;


/**
 * @ClassName:       excel导入工具类
 * @Description:
 * @author:  wangyun
 * @date: 2018-3-20     上午10:27:52
 */
public class ImportExcelUtil {
    private Workbook wb;
    private Sheet sheet;
    private Row row;


    public ImportExcelUtil(String filepath) {
        if(filepath==null){
            return;
        }
        String ext = filepath.substring(filepath.lastIndexOf("."));//获取文件名后缀
        try {
            InputStream is = new FileInputStream(filepath);
            if(".xls".equals(ext)){//2003- 版本的excel
                wb = new HSSFWorkbook(is);
            }else if(".xlsx".equals(ext)){//2007+ 版本的excel
                wb = new XSSFWorkbook(is);
            }else{
                wb=null;
            }
        } catch ( FileNotFoundException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }
    /**
     * 读取excel表头的内容
     * @return
     * @throws Exception
     */
    public String[] readExcelTitle() throws Exception{
        if(wb==null){
            throw new Exception("Workbook对象为空！");
        }
        sheet = wb.getSheetAt(0);
        row = sheet.getRow(0);

        // 标题总列数
        int colNum = row.getPhysicalNumberOfCells();
        System.out.println("colNum:" + colNum);
        String[] title = new String[colNum];
        for (int i = 0; i < colNum; i++) {
            // title[i] = getStringCellValue(row.getCell((short) i));
            row.getCell(i).setCellType(Cell.CELL_TYPE_STRING);
            title[i] = row.getCell(i).getStringCellValue();
        }
        return title;
    }
    /**
     * 读取excel内容
     * @return
     * @throws Exception
     */
    public JSONArray readExcelContent() throws Exception{
        if(wb==null){
            throw new Exception("Workbook对象为空！");
        }
        JSONArray jsonArray=new JSONArray();
        sheet = wb.getSheetAt(0);
        // 得到总行数
        int rowNum = sheet.getLastRowNum();
        row = sheet.getRow(0);
        int colNum = row.getPhysicalNumberOfCells();
        // 正文内容应该从第二行开始,第一行为表头的标题
        for (int i = 2; i <= rowNum; i++) {
            row = sheet.getRow(i);
            int j = 0;
            JSONObject jsonObject=new JSONObject();
            while (j < colNum) {
                Object obj = getCellFormatValue(row.getCell(j));
                Object objStr=getCellFormatValue(sheet.getRow(1).getCell(j));
                //获取第一列的标题
                if (!"yewucengji".equals(obj)){
                    String objStrNew=(String) objStr;
                    jsonObject.put(objStrNew.substring(objStrNew.indexOf(",")+1), obj);
                }else {
                    jsonObject.put((String) objStr, obj);
                }
                j++;
            }
            jsonArray.add(jsonObject);
        }
        return jsonArray;

    }
    /**
     * 根据cell类型设置数据参数
     * @param cell
     * @return
     */
    private Object getCellFormatValue(Cell cell) {
        Object cellvalue = "";
        if (cell != null) {
            cell.setCellType(Cell.CELL_TYPE_STRING);
            cellvalue = cell.getRichStringCellValue().getString();
        } else {
            cellvalue = "";
        }
        return cellvalue;

    }

    public static void main(String[] args) {
        String filepath = "C:\\总结与计划\\template_host (5).xlsx";
        ImportExcelUtil eu=new ImportExcelUtil(filepath);
        String[] title;
        try {
            title = eu.readExcelTitle();
		/*	System.out.println("获得Excel表格的标题:");
	    	for (String s : title) {
	    	System.out.print(s + " ");
	    	}*/

            // 对读取Excel表格内容测试
            JSONArray map = eu.readExcelContent();
            System.out.println("-------------------------------------");
            System.out.println("获得Excel表格的内容:"+map);
            System.out.println("-------------------------------------");
//            for (int i = 0; i <= map.size(); i++) {
//                System.out.println(map.get(i));
//            }
        } catch (FileNotFoundException e) {
            System.out.println("未找到指定路径的文件!");
            e.printStackTrace();
        }catch (Exception e) {
            e.printStackTrace();
        }



    }


}
